package thread;

import java.util.concurrent.PriorityBlockingQueue;

/**
 * 优先级阻塞队列
 */
// 表示一个任务
class MyTask implements Comparable<MyTask> {
    public Runnable runnable;
    public long time;


    public MyTask(Runnable runnable, long delay) {
        this.runnable = runnable;
        // 取当前时刻的时间戳 + delay ，作为该任务实际执行的时间戳
        this.time = System.currentTimeMillis() + delay;
    }

    // 意味着每次取出的是时间的最小元素
    @Override
    public int compareTo(MyTask o) {
        return (int) (this.time - o.time);
    }
}

class MyTimer {
    // 这个结构，带有优先级的阻塞队列，核心数据结构
    private static PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    private static Object locker = new Object();


    // 此处的 delay 是一个形如 3000 这样的数字（多长时间之后，执行该任务）
    public void schedule(Runnable runnable, long delay) {
        // 根据参数，构造 MyTask，插入列队即可
        MyTask myTask = new MyTask(runnable, delay);
        queue.put(myTask);
        synchronized (locker) {
            locker.notify();
        }
    }

    // 在这里构造线程，负责执行具体任务
    public MyTimer() {
        Thread t = new Thread(() -> {
            while (true) {
                try {
                    // 阻塞队列，只有阻塞的入队列和阻塞的出队列，没有阻塞的查看队首元素
                    synchronized (locker) {
                        MyTask myTask = queue.take();
                        long curTime = System.currentTimeMillis();  // 当前时刻
                        if (myTask.time <= curTime) {
                            // 时间到了
                            myTask.runnable.run();
                        } else {
                            // 时间没到
                            // 把刚才取出的任务，重新塞回队列中
                            queue.put(myTask);
                            locker.wait(myTask.time - curTime);   // 此处不加，会造成忙等，如果加sleep，可能会错过某个新的时间更早的任务
                        }
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        t.start();
    }


}

public class Demo19 {

    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        MyTimer myTimer1 = new MyTimer();

        myTimer1.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello4");
            }
        },4000);
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello5");
            }
        }, 5000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello3");
            }
        }, 3000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello2");
            }
        }, 2000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello1");
            }
        }, 1000);

        System.out.println("hello0");
    }

}
